################### bilstm-sbt
# bilstm-sbt-train:
CUDA_VISIBLE_DEVICES=5 python -m tasks.ClassSummary.models.BiLstm.main \
  --save_path ./model/bilstm-sbt/ \
  --input_flag sbt \
  --train ./tasks/ClassSummary/resource/train.token.jsonl \
  --val ./tasks/ClassSummary/resource/valid.token.jsonl \
  --code_field ./tasks/ClassSummary/resource/sbt.field \
  --nl_field ./tasks/ClassSummary/resource/nl.field \
  --patience 4 \
  --batch_size 64 \
  --max_epoch 100 \
  --lr 0.001 \
  --max_grad 5 \
  --gpu_num 1 \
  --seed 1024 \
  --check_freq 1

# bilstm-sbt-hybrid-test:
CUDA_VISIBLE_DEVICES=5 python -m tasks.ClassSummary.models.BiLstm.test \
  --input_flag sbt \
  --test ./tasks/ClassSummary/resource/test.token.jsonl \
  --code_field ./tasks/ClassSummary/resource/sbt.field \
  --nl_field ./tasks/ClassSummary/resource/nl.field \
  --check_point /home/hlcs/lmc/clsum/model/bilstm-sbt/BiLstm-sbt-epoch=28-bleu=10.8247.ckpt \
  --save_path ./translate/bilstm-sbt-hybrid.jsonl \
  --gpu_num 1 \
  --batch_size 64 \
  --seed 1024

# bilstm-sbt-cross-test:
CUDA_VISIBLE_DEVICES=5 python -m tasks.ClassSummary.models.BiLstm.test \
  --input_flag sbt \
  --test ./tasks/ClassSummary/resource/cross.token.jsonl \
  --code_field ./tasks/ClassSummary/resource/sbt.field \
  --nl_field ./tasks/ClassSummary/resource/nl.field \
  --check_point /home/hlcs/lmc/clsum/model/bilstm-sbt/BiLstm-sbt-epoch=28-bleu=10.8247.ckpt \
  --save_path ./translate/bilstm-sbt-cross.jsonl \
  --gpu_num 1 \
  --batch_size 64 \
  --seed 1024
